//package com.ffcs.oss.starter.websocket.config;
//
//import cn.iocoder.yudao.framework.mq.redis.config.YudaoRedisMQConsumerAutoConfiguration;
//import cn.iocoder.yudao.framework.mq.redis.core.RedisMQTemplate;
//import cn.iocoder.yudao.framework.websocket.core.handler.JsonWebSocketMessageHandler;
//import cn.iocoder.yudao.framework.websocket.core.listener.WebSocketMessageListener;
//import cn.iocoder.yudao.framework.websocket.core.security.LoginUserHandshakeInterceptor;
//import cn.iocoder.yudao.framework.websocket.core.security.WebSocketAuthorizeRequestsCustomizer;
//import cn.iocoder.yudao.framework.websocket.core.sender.kafka.KafkaWebSocketMessageConsumer;
//import cn.iocoder.yudao.framework.websocket.core.sender.kafka.KafkaWebSocketMessageSender;
//import cn.iocoder.yudao.framework.websocket.core.sender.local.LocalWebSocketMessageSender;
//import cn.iocoder.yudao.framework.websocket.core.sender.rabbitmq.RabbitMQWebSocketMessageConsumer;
//import cn.iocoder.yudao.framework.websocket.core.sender.rabbitmq.RabbitMQWebSocketMessageSender;
//import cn.iocoder.yudao.framework.websocket.core.sender.redis.RedisWebSocketMessageConsumer;
//import cn.iocoder.yudao.framework.websocket.core.sender.redis.RedisWebSocketMessageSender;
//import cn.iocoder.yudao.framework.websocket.core.sender.rocketmq.RocketMQWebSocketMessageConsumer;
//import cn.iocoder.yudao.framework.websocket.core.sender.rocketmq.RocketMQWebSocketMessageSender;
//import cn.iocoder.yudao.framework.websocket.core.session.WebSocketSessionHandlerDecorator;
//import cn.iocoder.yudao.framework.websocket.core.session.WebSocketSessionManager;
//import cn.iocoder.yudao.framework.websocket.core.session.WebSocketSessionManagerImpl;
//import org.apache.rocketmq.spring.core.RocketMQTemplate;
//import org.springframework.amqp.core.TopicExchange;
//import org.springframework.amqp.rabbit.core.RabbitTemplate;
//import org.springframework.beans.factory.annotation.Value;
//import org.springframework.boot.autoconfigure.AutoConfiguration;
//import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
//import org.springframework.boot.context.properties.EnableConfigurationProperties;
//import org.springframework.context.annotation.Bean;
//import org.springframework.context.annotation.Configuration;
//import org.springframework.kafka.core.KafkaTemplate;
//import org.springframework.web.socket.WebSocketHandler;
//import org.springframework.web.socket.config.annotation.EnableWebSocket;
//import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
//import org.springframework.web.socket.server.HandshakeInterceptor;
//
//import java.util.List;
//
///**
// * WebSocket 自动配置
// *
// * @author xingyu4j
// */
//@AutoConfiguration(before = YudaoRedisMQConsumerAutoConfiguration.class) // before YudaoRedisMQConsumerAutoConfiguration 的原因是，需要保证 RedisWebSocketMessageConsumer 先创建，才能创建 RedisMessageListenerContainer
//@EnableWebSocket // 开启 websocket
//@ConditionalOnProperty(prefix = "yudao.websocket", value = "enable", matchIfMissing = true) // 允许使用 yudao.websocket.enable=false 禁用 websocket
//@EnableConfigurationProperties(WebSocketProperties.class)
//public class YudaoWebSocketAutoConfiguration {
//
//    @Bean
//    public WebSocketConfigurer webSocketConfigurer(HandshakeInterceptor[] handshakeInterceptors,
//                                                   WebSocketHandler webSocketHandler,
//                                                   WebSocketProperties webSocketProperties) {
//        return registry -> registry
//                // 添加 WebSocketHandler
//                .addHandler(webSocketHandler, webSocketProperties.getPath())
//                .addInterceptors(handshakeInterceptors)
//                // 允许跨域，否则前端连接会直接断开
//                .setAllowedOriginPatterns("*");
//    }
//
//    @Bean
//    public HandshakeInterceptor handshakeInterceptor() {
//        return new LoginUserHandshakeInterceptor();
//    }
//
//    @Bean
//    public WebSocketHandler webSocketHandler(WebSocketSessionManager sessionManager,
//                                             List<? extends WebSocketMessageListener<?>> messageListeners) {
//        // 1. 创建 JsonWebSocketMessageHandler 对象，处理消息
//        JsonWebSocketMessageHandler messageHandler = new JsonWebSocketMessageHandler(messageListeners);
//        // 2. 创建 WebSocketSessionHandlerDecorator 对象，处理连接
//        return new WebSocketSessionHandlerDecorator(messageHandler, sessionManager);
//    }
//
//    @Bean
//    public WebSocketSessionManager webSocketSessionManager() {
//        return new WebSocketSessionManagerImpl();
//    }
//
//    @Bean
//    public WebSocketAuthorizeRequestsCustomizer webSocketAuthorizeRequestsCustomizer(WebSocketProperties webSocketProperties) {
//        return new WebSocketAuthorizeRequestsCustomizer(webSocketProperties);
//    }
//
//    // ==================== Sender 相关 ====================
//
//    @Configuration
//    @ConditionalOnProperty(prefix = "yudao.websocket", name = "sender-type", havingValue = "local")
//    public class LocalWebSocketMessageSenderConfiguration {
//
//        @Bean
//        public LocalWebSocketMessageSender localWebSocketMessageSender(WebSocketSessionManager sessionManager) {
//            return new LocalWebSocketMessageSender(sessionManager);
//        }
//
//    }
//
//    @Configuration
//    @ConditionalOnProperty(prefix = "yudao.websocket", name = "sender-type", havingValue = "redis")
//    public class RedisWebSocketMessageSenderConfiguration {
//
//        @Bean
//        public RedisWebSocketMessageSender redisWebSocketMessageSender(WebSocketSessionManager sessionManager,
//                                                                       RedisMQTemplate redisMQTemplate) {
//            return new RedisWebSocketMessageSender(sessionManager, redisMQTemplate);
//        }
//
//        @Bean
//        public RedisWebSocketMessageConsumer redisWebSocketMessageConsumer(
//                RedisWebSocketMessageSender redisWebSocketMessageSender) {
//            return new RedisWebSocketMessageConsumer(redisWebSocketMessageSender);
//        }
//
//    }
//
//    @Configuration
//    @ConditionalOnProperty(prefix = "yudao.websocket", name = "sender-type", havingValue = "rocketmq")
//    public class RocketMQWebSocketMessageSenderConfiguration {
//
//        @Bean
//        public RocketMQWebSocketMessageSender rocketMQWebSocketMessageSender(
//                WebSocketSessionManager sessionManager, RocketMQTemplate rocketMQTemplate,
//                @Value("${yudao.websocket.sender-rocketmq.topic}") String topic) {
//            return new RocketMQWebSocketMessageSender(sessionManager, rocketMQTemplate, topic);
//        }
//
//        @Bean
//        public RocketMQWebSocketMessageConsumer rocketMQWebSocketMessageConsumer(
//                RocketMQWebSocketMessageSender rocketMQWebSocketMessageSender) {
//            return new RocketMQWebSocketMessageConsumer(rocketMQWebSocketMessageSender);
//        }
//
//    }
//
//    @Configuration
//    @ConditionalOnProperty(prefix = "yudao.websocket", name = "sender-type", havingValue = "rabbitmq")
//    public class RabbitMQWebSocketMessageSenderConfiguration {
//
//        @Bean
//        public RabbitMQWebSocketMessageSender rabbitMQWebSocketMessageSender(
//                WebSocketSessionManager sessionManager, RabbitTemplate rabbitTemplate,
//                TopicExchange websocketTopicExchange) {
//            return new RabbitMQWebSocketMessageSender(sessionManager, rabbitTemplate, websocketTopicExchange);
//        }
//
//        @Bean
//        public RabbitMQWebSocketMessageConsumer rabbitMQWebSocketMessageConsumer(
//                RabbitMQWebSocketMessageSender rabbitMQWebSocketMessageSender) {
//            return new RabbitMQWebSocketMessageConsumer(rabbitMQWebSocketMessageSender);
//        }
//
//        /**
//         * 创建 Topic Exchange
//         */
//        @Bean
//        public TopicExchange websocketTopicExchange(@Value("${yudao.websocket.sender-rabbitmq.exchange}") String exchange) {
//            return new TopicExchange(exchange,
//                    true,  // durable: 是否持久化
//                    false);  // exclusive: 是否排它
//        }
//
//    }
//
//    @Configuration
//    @ConditionalOnProperty(prefix = "yudao.websocket", name = "sender-type", havingValue = "kafka")
//    public class KafkaWebSocketMessageSenderConfiguration {
//
//        @Bean
//        public KafkaWebSocketMessageSender kafkaWebSocketMessageSender(
//                WebSocketSessionManager sessionManager, KafkaTemplate<Object, Object> kafkaTemplate,
//                @Value("${yudao.websocket.sender-kafka.topic}") String topic) {
//            return new KafkaWebSocketMessageSender(sessionManager, kafkaTemplate, topic);
//        }
//
//        @Bean
//        public KafkaWebSocketMessageConsumer kafkaWebSocketMessageConsumer(
//                KafkaWebSocketMessageSender kafkaWebSocketMessageSender) {
//            return new KafkaWebSocketMessageConsumer(kafkaWebSocketMessageSender);
//        }
//
//    }
//
//}